# DATA
dataset='CULane'
data_root = 'D:/dataset/lane/ufld'

input_h = 288
input_w = 512

# TRAIN
epoch = 1000
batch_size = 2
optimizer = 'Adam'  #['SGD','Adam']
learning_rate = 0.015
weight_decay = 1e-4
momentum = 0.9

scheduler = 'multi' #['multi', 'cos']
steps = [25,38]
gamma  = 0.1
warmup = 'linear'
warmup_iters = 695

# NETWORK
use_aux = False
griding_num = 200
backbone = '34'

# LOSS
sim_loss_w = 0
shp_loss_w = 0

# EXP
note = ''

log_path = 'E:/kpts'

# FINETUNE or RESUME MODEL PATH
finetune = None
resume = None

# TEST
test_model = 'E:/kpts/20231220_104843_lr_3e-02_b_4/best.pth'
test_work_dir = 'D:/PycharmProject/Ultra-Fast-Lane-Detection/tmp'

num_lanes = 10




